#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from dragon.selector import Selector
from dragon.page import DraPage


class NBSe:
    naviBar = Selector('div.nav-tabbar', '导航栏')
    backTopButton = Selector('text=回到顶部', '回到顶部按钮')
    personalButton = Selector('text=个人中心', '个人中心按钮')
    mainForceButton = Selector('text=主力轨迹', '主力轨迹按钮')
    stockTrackButton = Selector('text=个股跟踪', '个股跟踪按钮')
    earlyTradeButton = Selector('text=早盘龙虎', '早盘龙虎按钮')
    emotionalCycleButton = Selector('text=情绪周期', '情绪周期按钮')
    subscribeButton = Selector('text=我的自选', '我的自选按钮')


class NaviBar:
    def __init__(self,draPage:DraPage):
        self.draPage = draPage
        self.naviBar = draPage.locator(NBSe.naviBar)
        self.backTopButton = draPage.locator(NBSe.backTopButton)
        self.personalButton = draPage.locator(NBSe.personalButton)
        self.mainForceButton = draPage.locator(NBSe.mainForceButton)
        self.stockTrackButton = draPage.locator(NBSe.stockTrackButton)
        self.earlyTradeButton = draPage.locator(NBSe.earlyTradeButton)
        self.emotionalCycleButton = draPage.locator(NBSe.emotionalCycleButton)
        self.subscribeButton=draPage.locator(NBSe.subscribeButton)


